# 使用 Python 3.6 的官方镜像作为基础镜像
FROM python:3.6

# 设置工作目录
WORKDIR /app

# 复制 rknn 工具链文件到容器中
COPY rknn_toolkit-1.6.0-cp36-cp36m-linux_x86_64.whl .

# 安装 rknn-toolkit
RUN pip install rknn_toolkit-1.6.0-cp36-cp36m-linux_x86_64.whl

# 如果有其他依赖，可以创建 requirements.txt 并安装
COPY requirements.txt .
RUN pip install -r requirements.txt

# 设置环境变量（如果需要的话）
# ENV VARIABLE_NAME=value

# 将当前目录挂载为工作目录（在 docker run 时使用）
VOLUME ["/app"]

# 容器启动时的默认命令
CMD ["python3"] 